package com.jhy.szl.daima20240816;

import java.util.HashSet;

/**
 * 编写一个算法来判断一个数 n 是不是快乐数。
 */

public class IsHappy {
    public boolean isHappy(int n) {
        HashSet<Integer> set = new HashSet<>();
        // 等于1 或者 遇到重复的值退出
        while (n != 1 && !set.contains(n)) {
            set.add(n);
            n = getNext(n);
        }
        return n == 1;
    }

    /**
     * 算每一个数的位数的平方
     *
     * @param n
     * @return
     */
    public int getNext(int n) {
        int sum = 0;
        while (n > 0) {
            int temp = n % 10;
            sum += temp * temp;
            // 取下一个数
            n = n / 10;
        }
        return sum;
    }

}
